# 数组的前几项依次累加：
# 例如：
# arr = [1, 2, 3, 4, 5, 6]
# 结果：res = [1, 3, 6, 10, 15, 21]
#
# 方法一：for循环
# 直接使用for循环来计算，这个不推荐，代码不展示
#
# 方法二：accumulate

from itertools import accumulate
arr = [1, 2, 3, 4, 5, 6]
res = list(accumulate(arr))
print(res)
# [1, 3, 6, 10, 15, 21]


# 拓展：
# accumulate可以传入函数

from itertools import accumulate
arr = [1, 2, 3, 4, 5, 6]
def fun(a, b):
    # 乘法函数                                                                                                      
    return a*b

res1 = list(accumulate(arr,fun))                   

print(res1)
# [1, 2, 6, 24, 120, 720]